#ifndef AUDIO_PROCESSOR_H
#define AUDIO_PROCESSOR_H

// 包含必要的头文件
#include <esp_afe_sr_models.h>  // ESP32音频前端和语音识别模型
#include <freertos/FreeRTOS.h>  // FreeRTOS操作系统
#include <freertos/task.h>      // FreeRTOS任务管理
#include <freertos/event_groups.h>  // FreeRTOS事件组

#include <string>
#include <vector>
#include <functional>

#include "audio_codec.h"  // 音频编解码器

// 音频处理器类
class AudioProcessor {
public:
    AudioProcessor();  // 构造函数
    ~AudioProcessor(); // 析构函数

    // 初始化音频处理器
    void Initialize(AudioCodec* codec, bool realtime_chat);
    
    // 输入音频数据
    void Feed(const std::vector<int16_t>& data);
    
    // 启动音频处理
    void Start();
    
    // 停止音频处理
    void Stop();
    
    // 检查是否正在运行
    bool IsRunning();
    
    // 设置音频输出回调函数
    void OnOutput(std::function<void(std::vector<int16_t>&& data)> callback);
    
    // 设置语音活动检测状态变化回调函数
    void OnVadStateChange(std::function<void(bool speaking)> callback);
    
    // 获取输入数据大小
    size_t GetFeedSize();

private:
    EventGroupHandle_t event_group_ = nullptr;  // FreeRTOS事件组句柄
    esp_afe_sr_iface_t* afe_iface_ = nullptr;  // 音频前端接口
    esp_afe_sr_data_t* afe_data_ = nullptr;    // 音频前端数据
    std::function<void(std::vector<int16_t>&& data)> output_callback_;  // 音频输出回调函数
    std::function<void(bool speaking)> vad_state_change_callback_;      // 语音活动检测状态变化回调函数
    AudioCodec* codec_ = nullptr;  // 音频编解码器实例
    bool is_speaking_ = false;     // 当前是否在说话

    // 音频处理任务
    void AudioProcessorTask();
};

#endif
